package com.ssm.controller;

import com.ssm.domain.SysLog;
import com.ssm.service.ISyslogService;
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.annotation.After;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.security.core.context.SecurityContext;
import org.springframework.security.core.context.SecurityContextHolder;
import org.springframework.security.core.userdetails.User;
import org.springframework.stereotype.Component;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;

import javax.servlet.http.HttpServletRequest;
import java.lang.reflect.Method;
import java.util.Date;

@Component
@Aspect
public class LogAop {

    @Autowired
    private ISyslogService syslogService;

    private Date visitTime;
    private Class clazz;
    private Method method;

    //前置通知 主要是获取开始时间，执行的类，执行的那哪个方法
    @Before("execution(* com.ssm.controller.*.*(..))")
    public void doBefore(JoinPoint jp) throws NoSuchMethodException {
        visitTime = new Date();                           //获取当前访问时间
        clazz = jp.getTarget().getClass();                //具体要访问的类
        String methodName = jp.getSignature().getName();  //获取访问的方法的名称
        Object[] args = jp.getArgs();                     //获取访问的方法的参数

        //获取具体执行的方法的method对象
        if(args==null||args.length==0){
            method = clazz.getMethod(methodName);         //只能获取无参数的方法
        }else{
            Class[] classArgs = new Class[args.length];
            for (int i =0 ;i<args.length;i++){
                classArgs[i] = args[i].getClass();
            }
            clazz.getMethod(methodName,classArgs);
        }
    }

    //后置通知
    @After("execution(* com.ssm.controller.*.*(..))")
    public void doAfter(JoinPoint jp){
        long time = new  Date().getTime() - visitTime.getTime();  //获取访问的时长

        String url = "";
        //获取url
        if (clazz!=null&&method!=null&&clazz != LogAop.class){
            //1.获取类上的@RequestMapping("orders")
            RequestMapping classAnnotation = (RequestMapping) clazz.getAnnotation(RequestMapping.class);
            if (classAnnotation != null){
                String[] classValue = classAnnotation.value();
                //2.获取方法上的@RequestMapping("xxx")
                RequestMapping methodAnnotation = method.getAnnotation(RequestMapping.class);
                if (methodAnnotation!=null){
                    String[] methodValue = methodAnnotation.value();
                    url = classValue[0]+methodValue[0];

                    //获取访问的ip
                    HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder.getRequestAttributes()).getRequest();
                    String ip = request.getRemoteAddr();

                    //获取访问的用户名
                    SecurityContext context = SecurityContextHolder.getContext();//从上下文中获得当前登录的用户
                    //SecurityContext context = request.getSession().getAttribute("SPRING_SECURITY_CONTEXT");
                    User user =(User) context.getAuthentication().getPrincipal();
                    String username = user.getUsername();

                    //把日志相关信息封装到SysLog对象
                    SysLog sysLog = new SysLog();
                    sysLog.setExecutionTime(time);//执行时长
                    sysLog.setIp(ip);
                    sysLog.setMethod("[类名] "+clazz.getName()+"[方法名] "+method.getName());
                    sysLog.setUrl(url);
                    sysLog.setUsername(username);
                    sysLog.setVisitTime(visitTime);

                    //调用service完成记录日志
                    syslogService.save(sysLog);
                }
            }
        }
    }
}
